import torch



class Config():
    def __init__(self):
        self.n_vocab=1002
        self.batch_size=256
        self.is_shuffle=True
        self.device=torch.device("cuda" if torch.cuda.is_available() else "cpu")
        self.embed_size=256
        self.hidden_size=256
        self.num_layers=1
        self.num_classes=2
        self.lr=0.0001
        self.num_epochs=10

